import configparser
import pymysql

# 创建 ConfigParser 对象
config = configparser.ConfigParser()

# 读取配置文件
config.read('../database_config.ini')
# 获取数据库连接配置信息
block = 'Database_test'  # 开发环境
# block = 'Database'      # 生产环境

host = config.get(block, 'host')
port = config.get(block, 'port')
username = config.get(block, 'username')
password = config.get(block, 'password')
db_name = config.get(block, 'database')


def getLinks():
    # 打开数据库连接
    db = pymysql.connect(host=host,
                         user=username,
                         password=password,
                         database=db_name)

    # 使用 cursor() 方法创建一个游标对象 cursor
    cursor = db.cursor()

    # 使用 execute()  方法执行 SQL 查询
    cursor.execute("SELECT link FROM list_urls order by id asc")

    data = cursor.fetchall()
    result = []
    for i in data:
        if i:
            result.append(i[0])

    # 关闭数据库连接
    db.close()

    return result


# for i in get_urls():
#     print(i)